Add some transition code and copied over some force stuff