Add return statement
Use exit instead