Augment Code

Really good codegen 

Unfortunately VSCode sucks and sometimes I have to manually clear the caches to get the plugin working again