Fixed bug

and added more examples for main