Re-fixed warm_up naming confusion and added docs.