Added comments functionality. Improved ui design