Polish a bit the code