Just have figured out what the "figure" fold for.

Restructure the folders and update my code a little bit.