Changed the "main()" a bit

hello