cut code down a little, fixed stage bonus