Working code now.