Updated my solution and reflection