Add tests for note and staff