Small changes pygmae init and set display