Adding tests to agent

Adding tests to agent