added more functionality and got partition working until dimension computations