fixed errors in code, wrong axis