Add extra NonNull annotations