added note feature and a test option, just need to clean up and add comments/readme file