add exceptions - fix bug