Removed broken import
Reformatted code