add count(*) / fix bugs / add tests