minor bug fix
Minor updates