Add unit test. Remove junk files.