Updated coded and fixed some axis labels