clean up code + learn GUI