Little re-factoring and code formatting.