added more code, almost done..just needs a couple more prints and some
error checking