Improves margins and flexbox styling