improve_week3_assigment