Added tests, must still work on README