Added drawing code for selection rect