tweaked problem set-up