Fixed main sketch to work again