#12: Added tests for state