Fixed some lines and removed unused variables.
Added punit as a submodule