adding update code to drain action

Signed-off-by: Jeffrey Peckham <peckham@pivotallabs.com>