skapat klasser, rensat skit. strukturerat om i "main"