Add version option

Feel free to rewrite as need be, just trying my hand at python