Added in custom code

Added code to stubs