Minor ML changes
merge + re-addition of if __name__==__main__ clause at the end
Added working UI list for categories, other misc. changes.