Create package and move code to __main__ module