Added code to investigate theta dependance