diff --git a/main.S b/main.S
index 2a9196a..4501b9c 100644
--- a/main.S
+++ b/main.S
@@ -94,16 +94,17 @@ init:
   ;; BIT1: output enable
   ;; BIT2: debug
   ;; BIT3: Signal waiting for data
-  ;; BIT6: UCB0SOMI - not used
-  mov.b   #BIT0|BIT1|BIT2|BIT3|BIT6,&P1DIR
+  mov.b   #BIT0|BIT1|BIT2|BIT3, &P1DIR
   mov.b   #0,&P1OUT
-  ;; BIT1: timer, short pulse
-  ;; BIT4: timer, long pulse
   ;; BIT5: spi, UCB0CLK
   ;; BIT6: spi, UCB0SOMI
   ;; BIT7: spi, UCB0SIMO
-  mov.b   #BIT1|BIT4|BIT5|BIT6|BIT7,&P2DIR
-  mov.b   #BIT1|BIT4|BIT5|BIT6|BIT7,&P2SEL
+  mov.b   #BIT5|BIT6|BIT7, &P1SEL
+  mov.b   #BIT5|BIT6|BIT7, &P1SEL2
+  ;; BIT1: timer, short pulse
+  ;; BIT4: timer, long pulse
+  mov.b   #BIT1|BIT4, &P2DIR
+  mov.b   #BIT1|BIT4, &P2SEL
 
   ;; timer configuration
   ;;  configure and stop timer
@@ -140,15 +141,15 @@ mainloop:
 
 mainloop_draw:
   call    #drawscreen
-  ;call    #forwardscreen
-
-  ;call    #wait
 
   ;; signal waiting for data
   set_signal_waiting_for_data
 
+  call    #forwardscreen
+  call    #wait
+
   ;; receive data via spi
-  call    #receivedata
+  ;call    #receivedata
 
   ;; data has been received, clear signal
   clear_signal_waiting_for_data