Added a display object and fixed issues