Learing python / if you got tips about my code write a comment in the code!