Adding back session check