Added more code. Finally getting somewhere!