Small changes to code

Commit so i can pull the latest version