#56 Add test