Added prob 1 and 2. Sort function broken