documentation + removed unecessary imports