Added gravity
Added partitional problem