Small tweak to styling of code on small screen