Minor change in drawRect logic