#40 Add test