Updated the code for css