plotting and better initiialisation