Code is currently working with all functionality