few modification in sample rake task