Final changes made to match outputs - working