reorganized main function, edited css, added button

Yeah.