Minor modifications
a little bit of formatting the code