Power unit upgrade