Moved custom layout code into own LayoutManager