fixes #2 remove bugs