my python code for practise