Added comments and train net instructions