added main, need to get working