Updated UI for commenting system