added classes and come more css styling