Small changes to code completed triv game