Just added a print, but that should be made better probably...