Sorting works now too!