Practice code for python