Added colors/removed unnecessary import
Changed formulation