Refact view for demo

this is going to crash because the indentation is wrong