simplified fista code