Clean up.
Added disconnect button + logic. [Does not work]