removed unnecessary packages and added some comments