Bug fixes, added --all flag