Add training/testing error over time graph. Also remove [h] from figures.