Minor additions. Removed a cheat i forgot to earlier.