Convert ui to package

Also moved initialization of pygame and video set mode to __init__ to
fix bugs