Added y axis label to graph and other small tweaks