Updated session check