Fixed some DBConnection stuff in classes