Improve css slightly