just removing some unnecessary layers