Add missing functionality to the solution

query google

Signed-off-by: Itai Koren <itkoren@gmail.com>