Fixes and added budget argument to throughput benchmark