Second homework, RSA generator in python