added the solution for week4b basic