still need to add readme and tweak code a bit