functionality working, need to render correctly