pep 8
add args to functions
complete function