Fixing small styling isues