adding files that make this thing work on aws gpu instance
some small changes