added a doctring, reorganized imports