bit of code cleaning.