Added save weights train extension