Removed unused code,

Added "if __name__ == __main__" statement.