Added flag. Now to make it look nicer.