Tryin' to fix the gui