Adding to inject functionality