debug the code/fix the bug: py; solved a little differently