Fixing styling issue a little bit