Mostly working

need to change aesthetics and start writing the paper!