update solution

updating solution to include reflection