Added small styles and some JS functions