Added get_returns function