Writing Multi Line Code In Python For Snake