fixing problems from pilot, plotting results