Still getting the graphics working correctly