Updates to necessary imports