added classes to make things more organized