Update controller logic with session