refactored model and edited README.