Writing Multi Line Code In Python For Prime