Added budget loop