updated to make run on ubuntu