Add generics to InvocationStage methods

Currently all of the InvocationStage methods don't have generics for the commands, which require casts and make it easier to reference a capture variable as it is just easier. We should add generics so that users can more easily get the type and have better compiler checks.