Add test file, bug fixes