Polishing tests and adding test module.