cleaned code, made it more readable (for me)