#5 add a test