Code for manuscript 'Nonlinear Ekman transport in balanced currents with curvature' by Wenegrat and Thomas.