added background color
fixed problems from class