Code fix

Fixed minor errors