bug fixes. nicer css.