Fix setup script
Hacky inital version on display.js