Update task2.py

now the masses are equilibrated