Optimized the code, rewritten unit test