Using consistent colours with other images
Adds optional ylim argument
Sets nicer formatter