added setDisplay function

Doesnt work, need to look into other things but the logic is now there