add tests... duh