Add agent.py with inheritance