Update lr to output gaussian parameters (for first part of project)