Added code guard so code displays properly in github.