[#71 WIP] add tests