started to add a class but don't know what I'm doing