Fixing styling issues with the challenge