Fixed small problems and added function for set color.