A bit of code reformatting - making it beautiful :)