We fixed it!

works properly now
update to UI

Implemented initialization function