Added comments & tweaked styling