tidied up the code a bit a ensured the cells are displaying in the demo app.