adding test, not fully working