Slight cleanup; removed unnecessary import