Testing the neural network on the agent

Right now the test is simple to get the box to always go forward on the
x-axis and zero on the y-axis. It hasn't been successful yet.