added ability to change colours, and line size