Added specifying hyperparameters of the modelling + doc updates