adding tests fixes #2