Fixed minor formatting issues of comments and a import error.