need more work on model