Update tests to both pass and test for adding notes