Added screen draw functionality