Added state prop