Fix the color issue and add ability to draw coordinate axes