Update p_OLS.py

Changed formatting to look leaner.