Small code change, input review DS