adding logic back in