Added notes as to what I want to improve