Add color

Should resolve #6