Enhance UI for like and comment feature