Input arguments added, PEP-8 formatting