added case for medium difficulty