updated according to expected output

some lil modification in output