Rewrite tests, and removal unnecessary print.