Finish initial solution of Kth largest element problem
Fix typos and small errors; add specs for partition helper method