Update controller logic