Main.hs improved but still buggy'cause I dunno how to start it