The dialog looks quite ugly right now, but at least everything works. I can work on making it prettier after this.
