Updates notes
Minor code cleanup