Fixing some small mistakes