Successfully fixed bug for assignment