Added missed reference
Minor fixes