Polished it a little bit