Add ability to chain some functions