Fixing get_version function + PEP8ing it