refactored js, updated css, added two buttons