Added some ui stuff and made main beautiful