Added comments to trainmodule