Please clean that up a bit by putting `axis = N + axis` on the next line. I know it wasn't yours to begin with ;)