added example inputs
still debugging algorithm
rewrote code a bit