Added LayoutManager calls but have a bug