Fixed heart bug

Hearts now go back even after first death