Semi working sortable feature (one bug left)