fixing small mistakes