fix readme and refactor models
fix readme and refactor models