cleaned up formatting

Signed-off-by: Arvind Chembarpu <achembarpu@gmail.com>
fixed is_prime check

Signed-off-by: Arvind Chembarpu <achembarpu@gmail.com>