added logic for review screen