Add sort on specific columns

Enable sorting on specific columns.