added solution for waffle001