Remove debug / unecassary code