Project 1 solution

Signed-off-by: whj0000 <robertwawnghaojia@gmail.com>