Add instance variable for color