Remove duplicate set_color function
Add test case for set_color function
Re-add missing parts
Merge pull request #783 from ahojnnes/draw

Remove set_color duplicate and add test case