added upgrade feature from entry-point