other changes to get working output from test file