Split up element.h into state.h and transition.h

Change-Id: I9867c3eb09d1b4be6cc668256cd8cd097558f291