Add main with working example
Some test changes