Modified assigment.sh with fixed bugs