Decompose forward function into initialize, predict, update (#105)

Decompose forward function into initialize, predict, update (#105) #325