Fix styling to be more robust

Not stuff up on smaller screens