More work on states

State rendering, switch loading state to separate JFrame maybe?