works, but fix single return number
works
works