little bug fixes and styling