[WIP] Add code to payload

Excuse the wrong branch name.