Adds missing imports