Add post build function