Improved styling. not yet responsive