Almost done with the layout, just have to make the sorting work