changed output slightly