Have the delegate and protocol working