Update solution

Signed-off-by: Maxence Charriere <maxence.charriere@outlook.com>