bug fix & remove unused files