Got first iteration working