Fixing styling issues raised