Fixing styling issues in the code