Fix std. Add passing tests for axis and big matrix