Adding brush code