Update DrawRect method