Removed hack.
TODO: plotting.