Added bonus question solution