little adjustments

removed unused import os
pep8 coding standard