Added imports
Added missing argument
Changed double to int
Added imports