small changes

Signed-off-by: Andrew R <andrewreds@gmail.com>