Adding finish code step