fixed bug and new figures