Add animation when notification is removed + some small fixes