Add some sorting functionality