Fixed styling to make it consistent across editing screens

Resolves #428