got rid of un-neccessary layer
removed unneccessary layer