Add ability to create notes + other improvements