Fixed styling the code