added test file, updated model