Spruced up the dockerfile