Update according to feedback
Initialize Theta