small changes to my addition. still just a stub for future code generator