Added come classes to style better