This method failed when second argument was not given. Slightly changed the API to always take a second argument without speed penalization and add optional argument axis. Added test.
