Added flux plotting capabilities