Fixing small mistakes