TST: add testing
TST: remove print stmts