Got the first iteration working