Add basic version of __.partition