Add figure dir

Add figure directory to store plots