Update initial solution - not finished