Updates to barn-agent.

@omidaladini