Polished some code