Improve graphs and start to make the parameters of the model a bit nicer.