Fixing weight issues in task 5.