Added drawing functionality