Added additional tests for code module + Polishing.