Adjust styling after code review