CSS update to improve UI