Update SS Lab Q8.a to take arguments

Signed-off-by: Karthik Nayak <karthik.188@gmail.com>