Added things to make it work