Fix selection sort.py

Add a proper function and add the sorting code in the function.

User `if __name__ == '__main__':` and write a few lines of code with a simple example.