Add logic for note controller, note model