My Python code while i am learning