#20 Add test