Remove outdated assertion
Tweak styles to match mockup