fix bug 3 and 4 task