Added test lines to verify programs works as intented