Add a drawRect implementation