Can you please add step by step documentation on setting your code up and running custom training job ?