add tests \o/