Cleaned up the design and css