fix bug, test exercise 4a