Revised dockerfile