Minor corrections in code