Add necessary State wrapper
Version 0.4.1