Removed matplotlib, should do plotting in separate file.