Extract method for moving and staying in penalty box

Signed-off-by: Igor Popov <igor.popov@napa.fi>