add new states for start and wait